# Copyright 2021 Wechat Group, Tencent
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

# Generated by the protocol buffer compiler.  DO NOT EDIT!
# source: tfcc_runtime/proto/operations/nn.proto

import sys

_b = sys.version_info[0] < 3 and (lambda x: x) or (lambda x: x.encode("latin1"))
from google.protobuf import descriptor as _descriptor
from google.protobuf import message as _message
from google.protobuf import reflection as _reflection
from google.protobuf import symbol_database as _symbol_database
from google.protobuf import descriptor_pb2

# @@protoc_insertion_point(imports)

_sym_db = _symbol_database.Default()


from .. import common_pb2 as tfcc__runtime_dot_proto_dot_common__pb2


DESCRIPTOR = _descriptor.FileDescriptor(
    name="tfcc_runtime/proto/operations/nn.proto",
    package="tfcc.runtime.operations.nn",
    syntax="proto3",
    serialized_pb=_b(
        '\n&tfcc_runtime/proto/operations/nn.proto\x12\x1atfcc.runtime.operations.nn\x1a\x1ftfcc_runtime/proto/common.proto"a\n\rAveragePool1D\x12\x0e\n\x06kernel\x18\x01 \x01(\r\x12\x0f\n\x07padding\x18\x02 \x01(\r\x12\x0e\n\x06stride\x18\x03 \x01(\r"\x1f\n\x07VERSION\x12\x05\n\x01_\x10\x00\x12\r\n\tVERSION_1\x10\x01"\xb9\x01\n\rAveragePool2D\x12\x15\n\rkernel_height\x18\x01 \x01(\r\x12\x14\n\x0ckernel_width\x18\x02 \x01(\r\x12\x16\n\x0epadding_height\x18\x03 \x01(\r\x12\x15\n\rpadding_width\x18\x04 \x01(\r\x12\x15\n\rstride_height\x18\x05 \x01(\r\x12\x14\n\x0cstride_width\x18\x06 \x01(\r"\x1f\n\x07VERSION\x12\x05\n\x01_\x10\x00\x12\r\n\tVERSION_1\x10\x01"\x7f\n\x12\x42\x61tchNormalization\x12+\n\x07\x65psilon\x18\x01 \x01(\x0b\x32\x1a.tfcc.runtime.common.Value\x12\x0c\n\x04\x61xis\x18\x02 \x01(\r".\n\x07VERSION\x12\x05\n\x01_\x10\x00\x12\r\n\tVERSION_1\x10\x01\x12\r\n\tVERSION_2\x10\x02"\xfc\x01\n\x06\x43onv2D\x12\x16\n\x0epadding_height\x18\x01 \x01(\r\x12\x15\n\rpadding_width\x18\x02 \x01(\r\x12\x15\n\rstride_height\x18\x03 \x01(\r\x12\x14\n\x0cstride_width\x18\x04 \x01(\r\x12\x15\n\rdilate_height\x18\x05 \x01(\r\x12\x14\n\x0c\x64ilate_width\x18\x06 \x01(\r\x12\x0c\n\x04nchw\x18\x07 \x01(\x08\x12\r\n\x05group\x18\x08 \x01(\r"L\n\x07VERSION\x12\x05\n\x01_\x10\x00\x12\r\n\tVERSION_1\x10\x01\x12\r\n\tVERSION_2\x10\x02\x12\r\n\tVERSION_3\x10\x03\x12\r\n\tVERSION_4\x10\x04"7\n\x06Gather\x12\x0c\n\x04\x61xis\x18\x01 \x01(\r"\x1f\n\x07VERSION\x12\x05\n\x01_\x10\x00\x12\r\n\tVERSION_1\x10\x01"6\n\x13GlobalAveragePool2D"\x1f\n\x07VERSION\x12\x05\n\x01_\x10\x00\x12\r\n\tVERSION_1\x10\x01"b\n\x12LayerNormalization\x12+\n\x07\x65psilon\x18\x01 \x01(\x0b\x32\x1a.tfcc.runtime.common.Value"\x1f\n\x07VERSION\x12\x05\n\x01_\x10\x00\x12\r\n\tVERSION_1\x10\x01"y\n\tMaxPool1D\x12\x0e\n\x06kernel\x18\x01 \x01(\r\x12\x0f\n\x07padding\x18\x02 \x01(\r\x12\x0e\n\x06stride\x18\x03 \x01(\r\x12\x0b\n\x03ncw\x18\x04 \x01(\x08".\n\x07VERSION\x12\x05\n\x01_\x10\x00\x12\r\n\tVERSION_1\x10\x01\x12\r\n\tVERSION_2\x10\x02"\xd2\x01\n\tMaxPool2D\x12\x15\n\rkernel_height\x18\x01 \x01(\r\x12\x14\n\x0ckernel_width\x18\x02 \x01(\r\x12\x16\n\x0epadding_height\x18\x03 \x01(\r\x12\x15\n\rpadding_width\x18\x04 \x01(\r\x12\x15\n\rstride_height\x18\x05 \x01(\r\x12\x14\n\x0cstride_width\x18\x06 \x01(\r\x12\x0c\n\x04nchw\x18\x07 \x01(\x08".\n\x07VERSION\x12\x05\n\x01_\x10\x00\x12\r\n\tVERSION_1\x10\x01\x12\r\n\tVERSION_2\x10\x02"*\n\x07NonZero"\x1f\n\x07VERSION\x12\x05\n\x01_\x10\x00\x12\r\n\tVERSION_1\x10\x01"\xc7\x01\n\x06OneHot\x12\r\n\x05\x64\x65pth\x18\x01 \x01(\r\x12-\n\toff_value\x18\x02 \x01(\x0b\x32\x1a.tfcc.runtime.common.Value\x12,\n\x08on_value\x18\x03 \x01(\x0b\x32\x1a.tfcc.runtime.common.Value\x12\x30\n\tdata_type\x18\x04 \x01(\x0e\x32\x1d.tfcc.runtime.common.DataType"\x1f\n\x07VERSION\x12\x05\n\x01_\x10\x00\x12\r\n\tVERSION_1\x10\x01"(\n\x05Where"\x1f\n\x07VERSION\x12\x05\n\x01_\x10\x00\x12\r\n\tVERSION_1\x10\x01"4\n\x11ScatterNDWithData"\x1f\n\x07VERSION\x12\x05\n\x01_\x10\x00\x12\r\n\tVERSION_1\x10\x01\x62\x06proto3'
    ),
    dependencies=[
        tfcc__runtime_dot_proto_dot_common__pb2.DESCRIPTOR,
    ],
)


_AVERAGEPOOL1D_VERSION = _descriptor.EnumDescriptor(
    name="VERSION",
    full_name="tfcc.runtime.operations.nn.AveragePool1D.VERSION",
    filename=None,
    file=DESCRIPTOR,
    values=[
        _descriptor.EnumValueDescriptor(
            name="_", index=0, number=0, options=None, type=None
        ),
        _descriptor.EnumValueDescriptor(
            name="VERSION_1", index=1, number=1, options=None, type=None
        ),
    ],
    containing_type=None,
    options=None,
    serialized_start=169,
    serialized_end=200,
)
_sym_db.RegisterEnumDescriptor(_AVERAGEPOOL1D_VERSION)

_AVERAGEPOOL2D_VERSION = _descriptor.EnumDescriptor(
    name="VERSION",
    full_name="tfcc.runtime.operations.nn.AveragePool2D.VERSION",
    filename=None,
    file=DESCRIPTOR,
    values=[
        _descriptor.EnumValueDescriptor(
            name="_", index=0, number=0, options=None, type=None
        ),
        _descriptor.EnumValueDescriptor(
            name="VERSION_1", index=1, number=1, options=None, type=None
        ),
    ],
    containing_type=None,
    options=None,
    serialized_start=169,
    serialized_end=200,
)
_sym_db.RegisterEnumDescriptor(_AVERAGEPOOL2D_VERSION)

_BATCHNORMALIZATION_VERSION = _descriptor.EnumDescriptor(
    name="VERSION",
    full_name="tfcc.runtime.operations.nn.BatchNormalization.VERSION",
    filename=None,
    file=DESCRIPTOR,
    values=[
        _descriptor.EnumValueDescriptor(
            name="_", index=0, number=0, options=None, type=None
        ),
        _descriptor.EnumValueDescriptor(
            name="VERSION_1", index=1, number=1, options=None, type=None
        ),
        _descriptor.EnumValueDescriptor(
            name="VERSION_2", index=2, number=2, options=None, type=None
        ),
    ],
    containing_type=None,
    options=None,
    serialized_start=471,
    serialized_end=517,
)
_sym_db.RegisterEnumDescriptor(_BATCHNORMALIZATION_VERSION)

_CONV2D_VERSION = _descriptor.EnumDescriptor(
    name="VERSION",
    full_name="tfcc.runtime.operations.nn.Conv2D.VERSION",
    filename=None,
    file=DESCRIPTOR,
    values=[
        _descriptor.EnumValueDescriptor(
            name="_", index=0, number=0, options=None, type=None
        ),
        _descriptor.EnumValueDescriptor(
            name="VERSION_1", index=1, number=1, options=None, type=None
        ),
        _descriptor.EnumValueDescriptor(
            name="VERSION_2", index=2, number=2, options=None, type=None
        ),
        _descriptor.EnumValueDescriptor(
            name="VERSION_3", index=3, number=3, options=None, type=None
        ),
        _descriptor.EnumValueDescriptor(
            name="VERSION_4", index=4, number=4, options=None, type=None
        ),
    ],
    containing_type=None,
    options=None,
    serialized_start=696,
    serialized_end=772,
)
_sym_db.RegisterEnumDescriptor(_CONV2D_VERSION)

_GATHER_VERSION = _descriptor.EnumDescriptor(
    name="VERSION",
    full_name="tfcc.runtime.operations.nn.Gather.VERSION",
    filename=None,
    file=DESCRIPTOR,
    values=[
        _descriptor.EnumValueDescriptor(
            name="_", index=0, number=0, options=None, type=None
        ),
        _descriptor.EnumValueDescriptor(
            name="VERSION_1", index=1, number=1, options=None, type=None
        ),
    ],
    containing_type=None,
    options=None,
    serialized_start=169,
    serialized_end=200,
)
_sym_db.RegisterEnumDescriptor(_GATHER_VERSION)

_GLOBALAVERAGEPOOL2D_VERSION = _descriptor.EnumDescriptor(
    name="VERSION",
    full_name="tfcc.runtime.operations.nn.GlobalAveragePool2D.VERSION",
    filename=None,
    file=DESCRIPTOR,
    values=[
        _descriptor.EnumValueDescriptor(
            name="_", index=0, number=0, options=None, type=None
        ),
        _descriptor.EnumValueDescriptor(
            name="VERSION_1", index=1, number=1, options=None, type=None
        ),
    ],
    containing_type=None,
    options=None,
    serialized_start=169,
    serialized_end=200,
)
_sym_db.RegisterEnumDescriptor(_GLOBALAVERAGEPOOL2D_VERSION)

_LAYERNORMALIZATION_VERSION = _descriptor.EnumDescriptor(
    name="VERSION",
    full_name="tfcc.runtime.operations.nn.LayerNormalization.VERSION",
    filename=None,
    file=DESCRIPTOR,
    values=[
        _descriptor.EnumValueDescriptor(
            name="_", index=0, number=0, options=None, type=None
        ),
        _descriptor.EnumValueDescriptor(
            name="VERSION_1", index=1, number=1, options=None, type=None
        ),
    ],
    containing_type=None,
    options=None,
    serialized_start=169,
    serialized_end=200,
)
_sym_db.RegisterEnumDescriptor(_LAYERNORMALIZATION_VERSION)

_MAXPOOL1D_VERSION = _descriptor.EnumDescriptor(
    name="VERSION",
    full_name="tfcc.runtime.operations.nn.MaxPool1D.VERSION",
    filename=None,
    file=DESCRIPTOR,
    values=[
        _descriptor.EnumValueDescriptor(
            name="_", index=0, number=0, options=None, type=None
        ),
        _descriptor.EnumValueDescriptor(
            name="VERSION_1", index=1, number=1, options=None, type=None
        ),
        _descriptor.EnumValueDescriptor(
            name="VERSION_2", index=2, number=2, options=None, type=None
        ),
    ],
    containing_type=None,
    options=None,
    serialized_start=471,
    serialized_end=517,
)
_sym_db.RegisterEnumDescriptor(_MAXPOOL1D_VERSION)

_MAXPOOL2D_VERSION = _descriptor.EnumDescriptor(
    name="VERSION",
    full_name="tfcc.runtime.operations.nn.MaxPool2D.VERSION",
    filename=None,
    file=DESCRIPTOR,
    values=[
        _descriptor.EnumValueDescriptor(
            name="_", index=0, number=0, options=None, type=None
        ),
        _descriptor.EnumValueDescriptor(
            name="VERSION_1", index=1, number=1, options=None, type=None
        ),
        _descriptor.EnumValueDescriptor(
            name="VERSION_2", index=2, number=2, options=None, type=None
        ),
    ],
    containing_type=None,
    options=None,
    serialized_start=471,
    serialized_end=517,
)
_sym_db.RegisterEnumDescriptor(_MAXPOOL2D_VERSION)

_NONZERO_VERSION = _descriptor.EnumDescriptor(
    name="VERSION",
    full_name="tfcc.runtime.operations.nn.NonZero.VERSION",
    filename=None,
    file=DESCRIPTOR,
    values=[
        _descriptor.EnumValueDescriptor(
            name="_", index=0, number=0, options=None, type=None
        ),
        _descriptor.EnumValueDescriptor(
            name="VERSION_1", index=1, number=1, options=None, type=None
        ),
    ],
    containing_type=None,
    options=None,
    serialized_start=169,
    serialized_end=200,
)
_sym_db.RegisterEnumDescriptor(_NONZERO_VERSION)

_ONEHOT_VERSION = _descriptor.EnumDescriptor(
    name="VERSION",
    full_name="tfcc.runtime.operations.nn.OneHot.VERSION",
    filename=None,
    file=DESCRIPTOR,
    values=[
        _descriptor.EnumValueDescriptor(
            name="_", index=0, number=0, options=None, type=None
        ),
        _descriptor.EnumValueDescriptor(
            name="VERSION_1", index=1, number=1, options=None, type=None
        ),
    ],
    containing_type=None,
    options=None,
    serialized_start=169,
    serialized_end=200,
)
_sym_db.RegisterEnumDescriptor(_ONEHOT_VERSION)

_WHERE_VERSION = _descriptor.EnumDescriptor(
    name="VERSION",
    full_name="tfcc.runtime.operations.nn.Where.VERSION",
    filename=None,
    file=DESCRIPTOR,
    values=[
        _descriptor.EnumValueDescriptor(
            name="_", index=0, number=0, options=None, type=None
        ),
        _descriptor.EnumValueDescriptor(
            name="VERSION_1", index=1, number=1, options=None, type=None
        ),
    ],
    containing_type=None,
    options=None,
    serialized_start=169,
    serialized_end=200,
)
_sym_db.RegisterEnumDescriptor(_WHERE_VERSION)

_SCATTERNDWITHDATA_VERSION = _descriptor.EnumDescriptor(
    name="VERSION",
    full_name="tfcc.runtime.operations.nn.ScatterNDWithData.VERSION",
    filename=None,
    file=DESCRIPTOR,
    values=[
        _descriptor.EnumValueDescriptor(
            name="_", index=0, number=0, options=None, type=None
        ),
        _descriptor.EnumValueDescriptor(
            name="VERSION_1", index=1, number=1, options=None, type=None
        ),
    ],
    containing_type=None,
    options=None,
    serialized_start=169,
    serialized_end=200,
)
_sym_db.RegisterEnumDescriptor(_SCATTERNDWITHDATA_VERSION)


_AVERAGEPOOL1D = _descriptor.Descriptor(
    name="AveragePool1D",
    full_name="tfcc.runtime.operations.nn.AveragePool1D",
    filename=None,
    file=DESCRIPTOR,
    containing_type=None,
    fields=[
        _descriptor.FieldDescriptor(
            name="kernel",
            full_name="tfcc.runtime.operations.nn.AveragePool1D.kernel",
            index=0,
            number=1,
            type=13,
            cpp_type=3,
            label=1,
            has_default_value=False,
            default_value=0,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
        _descriptor.FieldDescriptor(
            name="padding",
            full_name="tfcc.runtime.operations.nn.AveragePool1D.padding",
            index=1,
            number=2,
            type=13,
            cpp_type=3,
            label=1,
            has_default_value=False,
            default_value=0,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
        _descriptor.FieldDescriptor(
            name="stride",
            full_name="tfcc.runtime.operations.nn.AveragePool1D.stride",
            index=2,
            number=3,
            type=13,
            cpp_type=3,
            label=1,
            has_default_value=False,
            default_value=0,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
    ],
    extensions=[],
    nested_types=[],
    enum_types=[
        _AVERAGEPOOL1D_VERSION,
    ],
    options=None,
    is_extendable=False,
    syntax="proto3",
    extension_ranges=[],
    oneofs=[],
    serialized_start=103,
    serialized_end=200,
)


_AVERAGEPOOL2D = _descriptor.Descriptor(
    name="AveragePool2D",
    full_name="tfcc.runtime.operations.nn.AveragePool2D",
    filename=None,
    file=DESCRIPTOR,
    containing_type=None,
    fields=[
        _descriptor.FieldDescriptor(
            name="kernel_height",
            full_name="tfcc.runtime.operations.nn.AveragePool2D.kernel_height",
            index=0,
            number=1,
            type=13,
            cpp_type=3,
            label=1,
            has_default_value=False,
            default_value=0,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
        _descriptor.FieldDescriptor(
            name="kernel_width",
            full_name="tfcc.runtime.operations.nn.AveragePool2D.kernel_width",
            index=1,
            number=2,
            type=13,
            cpp_type=3,
            label=1,
            has_default_value=False,
            default_value=0,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
        _descriptor.FieldDescriptor(
            name="padding_height",
            full_name="tfcc.runtime.operations.nn.AveragePool2D.padding_height",
            index=2,
            number=3,
            type=13,
            cpp_type=3,
            label=1,
            has_default_value=False,
            default_value=0,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
        _descriptor.FieldDescriptor(
            name="padding_width",
            full_name="tfcc.runtime.operations.nn.AveragePool2D.padding_width",
            index=3,
            number=4,
            type=13,
            cpp_type=3,
            label=1,
            has_default_value=False,
            default_value=0,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
        _descriptor.FieldDescriptor(
            name="stride_height",
            full_name="tfcc.runtime.operations.nn.AveragePool2D.stride_height",
            index=4,
            number=5,
            type=13,
            cpp_type=3,
            label=1,
            has_default_value=False,
            default_value=0,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
        _descriptor.FieldDescriptor(
            name="stride_width",
            full_name="tfcc.runtime.operations.nn.AveragePool2D.stride_width",
            index=5,
            number=6,
            type=13,
            cpp_type=3,
            label=1,
            has_default_value=False,
            default_value=0,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
    ],
    extensions=[],
    nested_types=[],
    enum_types=[
        _AVERAGEPOOL2D_VERSION,
    ],
    options=None,
    is_extendable=False,
    syntax="proto3",
    extension_ranges=[],
    oneofs=[],
    serialized_start=203,
    serialized_end=388,
)


_BATCHNORMALIZATION = _descriptor.Descriptor(
    name="BatchNormalization",
    full_name="tfcc.runtime.operations.nn.BatchNormalization",
    filename=None,
    file=DESCRIPTOR,
    containing_type=None,
    fields=[
        _descriptor.FieldDescriptor(
            name="epsilon",
            full_name="tfcc.runtime.operations.nn.BatchNormalization.epsilon",
            index=0,
            number=1,
            type=11,
            cpp_type=10,
            label=1,
            has_default_value=False,
            default_value=None,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
        _descriptor.FieldDescriptor(
            name="axis",
            full_name="tfcc.runtime.operations.nn.BatchNormalization.axis",
            index=1,
            number=2,
            type=13,
            cpp_type=3,
            label=1,
            has_default_value=False,
            default_value=0,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
    ],
    extensions=[],
    nested_types=[],
    enum_types=[
        _BATCHNORMALIZATION_VERSION,
    ],
    options=None,
    is_extendable=False,
    syntax="proto3",
    extension_ranges=[],
    oneofs=[],
    serialized_start=390,
    serialized_end=517,
)


_CONV2D = _descriptor.Descriptor(
    name="Conv2D",
    full_name="tfcc.runtime.operations.nn.Conv2D",
    filename=None,
    file=DESCRIPTOR,
    containing_type=None,
    fields=[
        _descriptor.FieldDescriptor(
            name="padding_height",
            full_name="tfcc.runtime.operations.nn.Conv2D.padding_height",
            index=0,
            number=1,
            type=13,
            cpp_type=3,
            label=1,
            has_default_value=False,
            default_value=0,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
        _descriptor.FieldDescriptor(
            name="padding_width",
            full_name="tfcc.runtime.operations.nn.Conv2D.padding_width",
            index=1,
            number=2,
            type=13,
            cpp_type=3,
            label=1,
            has_default_value=False,
            default_value=0,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
        _descriptor.FieldDescriptor(
            name="stride_height",
            full_name="tfcc.runtime.operations.nn.Conv2D.stride_height",
            index=2,
            number=3,
            type=13,
            cpp_type=3,
            label=1,
            has_default_value=False,
            default_value=0,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
        _descriptor.FieldDescriptor(
            name="stride_width",
            full_name="tfcc.runtime.operations.nn.Conv2D.stride_width",
            index=3,
            number=4,
            type=13,
            cpp_type=3,
            label=1,
            has_default_value=False,
            default_value=0,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
        _descriptor.FieldDescriptor(
            name="dilate_height",
            full_name="tfcc.runtime.operations.nn.Conv2D.dilate_height",
            index=4,
            number=5,
            type=13,
            cpp_type=3,
            label=1,
            has_default_value=False,
            default_value=0,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
        _descriptor.FieldDescriptor(
            name="dilate_width",
            full_name="tfcc.runtime.operations.nn.Conv2D.dilate_width",
            index=5,
            number=6,
            type=13,
            cpp_type=3,
            label=1,
            has_default_value=False,
            default_value=0,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
        _descriptor.FieldDescriptor(
            name="nchw",
            full_name="tfcc.runtime.operations.nn.Conv2D.nchw",
            index=6,
            number=7,
            type=8,
            cpp_type=7,
            label=1,
            has_default_value=False,
            default_value=False,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
        _descriptor.FieldDescriptor(
            name="group",
            full_name="tfcc.runtime.operations.nn.Conv2D.group",
            index=7,
            number=8,
            type=13,
            cpp_type=3,
            label=1,
            has_default_value=False,
            default_value=0,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
    ],
    extensions=[],
    nested_types=[],
    enum_types=[
        _CONV2D_VERSION,
    ],
    options=None,
    is_extendable=False,
    syntax="proto3",
    extension_ranges=[],
    oneofs=[],
    serialized_start=520,
    serialized_end=772,
)


_GATHER = _descriptor.Descriptor(
    name="Gather",
    full_name="tfcc.runtime.operations.nn.Gather",
    filename=None,
    file=DESCRIPTOR,
    containing_type=None,
    fields=[
        _descriptor.FieldDescriptor(
            name="axis",
            full_name="tfcc.runtime.operations.nn.Gather.axis",
            index=0,
            number=1,
            type=13,
            cpp_type=3,
            label=1,
            has_default_value=False,
            default_value=0,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
    ],
    extensions=[],
    nested_types=[],
    enum_types=[
        _GATHER_VERSION,
    ],
    options=None,
    is_extendable=False,
    syntax="proto3",
    extension_ranges=[],
    oneofs=[],
    serialized_start=774,
    serialized_end=829,
)


_GLOBALAVERAGEPOOL2D = _descriptor.Descriptor(
    name="GlobalAveragePool2D",
    full_name="tfcc.runtime.operations.nn.GlobalAveragePool2D",
    filename=None,
    file=DESCRIPTOR,
    containing_type=None,
    fields=[],
    extensions=[],
    nested_types=[],
    enum_types=[
        _GLOBALAVERAGEPOOL2D_VERSION,
    ],
    options=None,
    is_extendable=False,
    syntax="proto3",
    extension_ranges=[],
    oneofs=[],
    serialized_start=831,
    serialized_end=885,
)


_LAYERNORMALIZATION = _descriptor.Descriptor(
    name="LayerNormalization",
    full_name="tfcc.runtime.operations.nn.LayerNormalization",
    filename=None,
    file=DESCRIPTOR,
    containing_type=None,
    fields=[
        _descriptor.FieldDescriptor(
            name="epsilon",
            full_name="tfcc.runtime.operations.nn.LayerNormalization.epsilon",
            index=0,
            number=1,
            type=11,
            cpp_type=10,
            label=1,
            has_default_value=False,
            default_value=None,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
    ],
    extensions=[],
    nested_types=[],
    enum_types=[
        _LAYERNORMALIZATION_VERSION,
    ],
    options=None,
    is_extendable=False,
    syntax="proto3",
    extension_ranges=[],
    oneofs=[],
    serialized_start=887,
    serialized_end=985,
)


_MAXPOOL1D = _descriptor.Descriptor(
    name="MaxPool1D",
    full_name="tfcc.runtime.operations.nn.MaxPool1D",
    filename=None,
    file=DESCRIPTOR,
    containing_type=None,
    fields=[
        _descriptor.FieldDescriptor(
            name="kernel",
            full_name="tfcc.runtime.operations.nn.MaxPool1D.kernel",
            index=0,
            number=1,
            type=13,
            cpp_type=3,
            label=1,
            has_default_value=False,
            default_value=0,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
        _descriptor.FieldDescriptor(
            name="padding",
            full_name="tfcc.runtime.operations.nn.MaxPool1D.padding",
            index=1,
            number=2,
            type=13,
            cpp_type=3,
            label=1,
            has_default_value=False,
            default_value=0,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
        _descriptor.FieldDescriptor(
            name="stride",
            full_name="tfcc.runtime.operations.nn.MaxPool1D.stride",
            index=2,
            number=3,
            type=13,
            cpp_type=3,
            label=1,
            has_default_value=False,
            default_value=0,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
        _descriptor.FieldDescriptor(
            name="ncw",
            full_name="tfcc.runtime.operations.nn.MaxPool1D.ncw",
            index=3,
            number=4,
            type=8,
            cpp_type=7,
            label=1,
            has_default_value=False,
            default_value=False,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
    ],
    extensions=[],
    nested_types=[],
    enum_types=[
        _MAXPOOL1D_VERSION,
    ],
    options=None,
    is_extendable=False,
    syntax="proto3",
    extension_ranges=[],
    oneofs=[],
    serialized_start=987,
    serialized_end=1108,
)


_MAXPOOL2D = _descriptor.Descriptor(
    name="MaxPool2D",
    full_name="tfcc.runtime.operations.nn.MaxPool2D",
    filename=None,
    file=DESCRIPTOR,
    containing_type=None,
    fields=[
        _descriptor.FieldDescriptor(
            name="kernel_height",
            full_name="tfcc.runtime.operations.nn.MaxPool2D.kernel_height",
            index=0,
            number=1,
            type=13,
            cpp_type=3,
            label=1,
            has_default_value=False,
            default_value=0,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
        _descriptor.FieldDescriptor(
            name="kernel_width",
            full_name="tfcc.runtime.operations.nn.MaxPool2D.kernel_width",
            index=1,
            number=2,
            type=13,
            cpp_type=3,
            label=1,
            has_default_value=False,
            default_value=0,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
        _descriptor.FieldDescriptor(
            name="padding_height",
            full_name="tfcc.runtime.operations.nn.MaxPool2D.padding_height",
            index=2,
            number=3,
            type=13,
            cpp_type=3,
            label=1,
            has_default_value=False,
            default_value=0,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
        _descriptor.FieldDescriptor(
            name="padding_width",
            full_name="tfcc.runtime.operations.nn.MaxPool2D.padding_width",
            index=3,
            number=4,
            type=13,
            cpp_type=3,
            label=1,
            has_default_value=False,
            default_value=0,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
        _descriptor.FieldDescriptor(
            name="stride_height",
            full_name="tfcc.runtime.operations.nn.MaxPool2D.stride_height",
            index=4,
            number=5,
            type=13,
            cpp_type=3,
            label=1,
            has_default_value=False,
            default_value=0,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
        _descriptor.FieldDescriptor(
            name="stride_width",
            full_name="tfcc.runtime.operations.nn.MaxPool2D.stride_width",
            index=5,
            number=6,
            type=13,
            cpp_type=3,
            label=1,
            has_default_value=False,
            default_value=0,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
        _descriptor.FieldDescriptor(
            name="nchw",
            full_name="tfcc.runtime.operations.nn.MaxPool2D.nchw",
            index=6,
            number=7,
            type=8,
            cpp_type=7,
            label=1,
            has_default_value=False,
            default_value=False,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
    ],
    extensions=[],
    nested_types=[],
    enum_types=[
        _MAXPOOL2D_VERSION,
    ],
    options=None,
    is_extendable=False,
    syntax="proto3",
    extension_ranges=[],
    oneofs=[],
    serialized_start=1111,
    serialized_end=1321,
)


_NONZERO = _descriptor.Descriptor(
    name="NonZero",
    full_name="tfcc.runtime.operations.nn.NonZero",
    filename=None,
    file=DESCRIPTOR,
    containing_type=None,
    fields=[],
    extensions=[],
    nested_types=[],
    enum_types=[
        _NONZERO_VERSION,
    ],
    options=None,
    is_extendable=False,
    syntax="proto3",
    extension_ranges=[],
    oneofs=[],
    serialized_start=1323,
    serialized_end=1365,
)


_ONEHOT = _descriptor.Descriptor(
    name="OneHot",
    full_name="tfcc.runtime.operations.nn.OneHot",
    filename=None,
    file=DESCRIPTOR,
    containing_type=None,
    fields=[
        _descriptor.FieldDescriptor(
            name="depth",
            full_name="tfcc.runtime.operations.nn.OneHot.depth",
            index=0,
            number=1,
            type=13,
            cpp_type=3,
            label=1,
            has_default_value=False,
            default_value=0,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
        _descriptor.FieldDescriptor(
            name="off_value",
            full_name="tfcc.runtime.operations.nn.OneHot.off_value",
            index=1,
            number=2,
            type=11,
            cpp_type=10,
            label=1,
            has_default_value=False,
            default_value=None,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
        _descriptor.FieldDescriptor(
            name="on_value",
            full_name="tfcc.runtime.operations.nn.OneHot.on_value",
            index=2,
            number=3,
            type=11,
            cpp_type=10,
            label=1,
            has_default_value=False,
            default_value=None,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
        _descriptor.FieldDescriptor(
            name="data_type",
            full_name="tfcc.runtime.operations.nn.OneHot.data_type",
            index=3,
            number=4,
            type=14,
            cpp_type=8,
            label=1,
            has_default_value=False,
            default_value=0,
            message_type=None,
            enum_type=None,
            containing_type=None,
            is_extension=False,
            extension_scope=None,
            options=None,
            file=DESCRIPTOR,
        ),
    ],
    extensions=[],
    nested_types=[],
    enum_types=[
        _ONEHOT_VERSION,
    ],
    options=None,
    is_extendable=False,
    syntax="proto3",
    extension_ranges=[],
    oneofs=[],
    serialized_start=1368,
    serialized_end=1567,
)


_WHERE = _descriptor.Descriptor(
    name="Where",
    full_name="tfcc.runtime.operations.nn.Where",
    filename=None,
    file=DESCRIPTOR,
    containing_type=None,
    fields=[],
    extensions=[],
    nested_types=[],
    enum_types=[
        _WHERE_VERSION,
    ],
    options=None,
    is_extendable=False,
    syntax="proto3",
    extension_ranges=[],
    oneofs=[],
    serialized_start=1569,
    serialized_end=1609,
)


_SCATTERNDWITHDATA = _descriptor.Descriptor(
    name="ScatterNDWithData",
    full_name="tfcc.runtime.operations.nn.ScatterNDWithData",
    filename=None,
    file=DESCRIPTOR,
    containing_type=None,
    fields=[],
    extensions=[],
    nested_types=[],
    enum_types=[
        _SCATTERNDWITHDATA_VERSION,
    ],
    options=None,
    is_extendable=False,
    syntax="proto3",
    extension_ranges=[],
    oneofs=[],
    serialized_start=1611,
    serialized_end=1663,
)

_AVERAGEPOOL1D_VERSION.containing_type = _AVERAGEPOOL1D
_AVERAGEPOOL2D_VERSION.containing_type = _AVERAGEPOOL2D
_BATCHNORMALIZATION.fields_by_name[
    "epsilon"
].message_type = tfcc__runtime_dot_proto_dot_common__pb2._VALUE
_BATCHNORMALIZATION_VERSION.containing_type = _BATCHNORMALIZATION
_CONV2D_VERSION.containing_type = _CONV2D
_GATHER_VERSION.containing_type = _GATHER
_GLOBALAVERAGEPOOL2D_VERSION.containing_type = _GLOBALAVERAGEPOOL2D
_LAYERNORMALIZATION.fields_by_name[
    "epsilon"
].message_type = tfcc__runtime_dot_proto_dot_common__pb2._VALUE
_LAYERNORMALIZATION_VERSION.containing_type = _LAYERNORMALIZATION
_MAXPOOL1D_VERSION.containing_type = _MAXPOOL1D
_MAXPOOL2D_VERSION.containing_type = _MAXPOOL2D
_NONZERO_VERSION.containing_type = _NONZERO
_ONEHOT.fields_by_name[
    "off_value"
].message_type = tfcc__runtime_dot_proto_dot_common__pb2._VALUE
_ONEHOT.fields_by_name[
    "on_value"
].message_type = tfcc__runtime_dot_proto_dot_common__pb2._VALUE
_ONEHOT.fields_by_name[
    "data_type"
].enum_type = tfcc__runtime_dot_proto_dot_common__pb2._DATATYPE
_ONEHOT_VERSION.containing_type = _ONEHOT
_WHERE_VERSION.containing_type = _WHERE
_SCATTERNDWITHDATA_VERSION.containing_type = _SCATTERNDWITHDATA
DESCRIPTOR.message_types_by_name["AveragePool1D"] = _AVERAGEPOOL1D
DESCRIPTOR.message_types_by_name["AveragePool2D"] = _AVERAGEPOOL2D
DESCRIPTOR.message_types_by_name["BatchNormalization"] = _BATCHNORMALIZATION
DESCRIPTOR.message_types_by_name["Conv2D"] = _CONV2D
DESCRIPTOR.message_types_by_name["Gather"] = _GATHER
DESCRIPTOR.message_types_by_name["GlobalAveragePool2D"] = _GLOBALAVERAGEPOOL2D
DESCRIPTOR.message_types_by_name["LayerNormalization"] = _LAYERNORMALIZATION
DESCRIPTOR.message_types_by_name["MaxPool1D"] = _MAXPOOL1D
DESCRIPTOR.message_types_by_name["MaxPool2D"] = _MAXPOOL2D
DESCRIPTOR.message_types_by_name["NonZero"] = _NONZERO
DESCRIPTOR.message_types_by_name["OneHot"] = _ONEHOT
DESCRIPTOR.message_types_by_name["Where"] = _WHERE
DESCRIPTOR.message_types_by_name["ScatterNDWithData"] = _SCATTERNDWITHDATA
_sym_db.RegisterFileDescriptor(DESCRIPTOR)

AveragePool1D = _reflection.GeneratedProtocolMessageType(
    "AveragePool1D",
    (_message.Message,),
    dict(
        DESCRIPTOR=_AVERAGEPOOL1D,
        __module__="tfcc_runtime.proto.operations.nn_pb2"
        # @@protoc_insertion_point(class_scope:tfcc.runtime.operations.nn.AveragePool1D)
    ),
)
_sym_db.RegisterMessage(AveragePool1D)

AveragePool2D = _reflection.GeneratedProtocolMessageType(
    "AveragePool2D",
    (_message.Message,),
    dict(
        DESCRIPTOR=_AVERAGEPOOL2D,
        __module__="tfcc_runtime.proto.operations.nn_pb2"
        # @@protoc_insertion_point(class_scope:tfcc.runtime.operations.nn.AveragePool2D)
    ),
)
_sym_db.RegisterMessage(AveragePool2D)

BatchNormalization = _reflection.GeneratedProtocolMessageType(
    "BatchNormalization",
    (_message.Message,),
    dict(
        DESCRIPTOR=_BATCHNORMALIZATION,
        __module__="tfcc_runtime.proto.operations.nn_pb2"
        # @@protoc_insertion_point(class_scope:tfcc.runtime.operations.nn.BatchNormalization)
    ),
)
_sym_db.RegisterMessage(BatchNormalization)

Conv2D = _reflection.GeneratedProtocolMessageType(
    "Conv2D",
    (_message.Message,),
    dict(
        DESCRIPTOR=_CONV2D,
        __module__="tfcc_runtime.proto.operations.nn_pb2"
        # @@protoc_insertion_point(class_scope:tfcc.runtime.operations.nn.Conv2D)
    ),
)
_sym_db.RegisterMessage(Conv2D)

Gather = _reflection.GeneratedProtocolMessageType(
    "Gather",
    (_message.Message,),
    dict(
        DESCRIPTOR=_GATHER,
        __module__="tfcc_runtime.proto.operations.nn_pb2"
        # @@protoc_insertion_point(class_scope:tfcc.runtime.operations.nn.Gather)
    ),
)
_sym_db.RegisterMessage(Gather)

GlobalAveragePool2D = _reflection.GeneratedProtocolMessageType(
    "GlobalAveragePool2D",
    (_message.Message,),
    dict(
        DESCRIPTOR=_GLOBALAVERAGEPOOL2D,
        __module__="tfcc_runtime.proto.operations.nn_pb2"
        # @@protoc_insertion_point(class_scope:tfcc.runtime.operations.nn.GlobalAveragePool2D)
    ),
)
_sym_db.RegisterMessage(GlobalAveragePool2D)

LayerNormalization = _reflection.GeneratedProtocolMessageType(
    "LayerNormalization",
    (_message.Message,),
    dict(
        DESCRIPTOR=_LAYERNORMALIZATION,
        __module__="tfcc_runtime.proto.operations.nn_pb2"
        # @@protoc_insertion_point(class_scope:tfcc.runtime.operations.nn.LayerNormalization)
    ),
)
_sym_db.RegisterMessage(LayerNormalization)

MaxPool1D = _reflection.GeneratedProtocolMessageType(
    "MaxPool1D",
    (_message.Message,),
    dict(
        DESCRIPTOR=_MAXPOOL1D,
        __module__="tfcc_runtime.proto.operations.nn_pb2"
        # @@protoc_insertion_point(class_scope:tfcc.runtime.operations.nn.MaxPool1D)
    ),
)
_sym_db.RegisterMessage(MaxPool1D)

MaxPool2D = _reflection.GeneratedProtocolMessageType(
    "MaxPool2D",
    (_message.Message,),
    dict(
        DESCRIPTOR=_MAXPOOL2D,
        __module__="tfcc_runtime.proto.operations.nn_pb2"
        # @@protoc_insertion_point(class_scope:tfcc.runtime.operations.nn.MaxPool2D)
    ),
)
_sym_db.RegisterMessage(MaxPool2D)

NonZero = _reflection.GeneratedProtocolMessageType(
    "NonZero",
    (_message.Message,),
    dict(
        DESCRIPTOR=_NONZERO,
        __module__="tfcc_runtime.proto.operations.nn_pb2"
        # @@protoc_insertion_point(class_scope:tfcc.runtime.operations.nn.NonZero)
    ),
)
_sym_db.RegisterMessage(NonZero)

OneHot = _reflection.GeneratedProtocolMessageType(
    "OneHot",
    (_message.Message,),
    dict(
        DESCRIPTOR=_ONEHOT,
        __module__="tfcc_runtime.proto.operations.nn_pb2"
        # @@protoc_insertion_point(class_scope:tfcc.runtime.operations.nn.OneHot)
    ),
)
_sym_db.RegisterMessage(OneHot)

Where = _reflection.GeneratedProtocolMessageType(
    "Where",
    (_message.Message,),
    dict(
        DESCRIPTOR=_WHERE,
        __module__="tfcc_runtime.proto.operations.nn_pb2"
        # @@protoc_insertion_point(class_scope:tfcc.runtime.operations.nn.Where)
    ),
)
_sym_db.RegisterMessage(Where)

ScatterNDWithData = _reflection.GeneratedProtocolMessageType(
    "ScatterNDWithData",
    (_message.Message,),
    dict(
        DESCRIPTOR=_SCATTERNDWITHDATA,
        __module__="tfcc_runtime.proto.operations.nn_pb2"
        # @@protoc_insertion_point(class_scope:tfcc.runtime.operations.nn.ScatterNDWithData)
    ),
)
_sym_db.RegisterMessage(ScatterNDWithData)


# @@protoc_insertion_point(module_scope)
